"""
Qwen3模型配置文件
"""

# 通用配置
HF_ENDPOINT = "https://hf-mirror.com"  # HuggingFace镜像
HF_HUB_DISABLE_XET = "1"  # 强制禁用Xet Storage

# Embedding模型配置
EMBEDDING_CONFIG = {
    "local_model_path": "D:/ProjectWorkspace/shellMakefileDeploymentProjects/windowsDeploymentProjects/hfd-download-project/models/Qwen3-Embedding-0.6B",  # 本地模型路径，如果为None则使用model_id从HuggingFace加载
    "model_id": "Qwen/Qwen3-Embedding-0.6B",  # 模型ID
    "use_flash_attention": False,  # 是否使用flash_attention_2加速
    "use_cross_encoder": False,  # 是否使用CrossEncoder加载方式
}

# Reranker模型配置
RERANK_CONFIG = {
    # "local_model_path": None, # 本地模型路径，如果为None则使用model_id从HuggingFace加载
    "local_model_path": "D:/ProjectWorkspace/shellMakefileDeploymentProjects/windowsDeploymentProjects/hfd-download-project/models/Qwen3-Reranker-0.6B",
    "model_id": "Qwen/Qwen3-Reranker-0.6B",  # 模型ID
    "use_flash_attention": False,  # 是否使用flash_attention_2加速
    "use_fp16": False,  # 是否使用FP16精度
    "use_cross_encoder": False,  # 是否使用CrossEncoder加载方式
}

# 示例数据
EXAMPLE_QUERIES = [
    "What is the capital of China?",
    "Explain gravity",
]

EXAMPLE_DOCUMENTS = [
    "The capital of China is Beijing.",
    "Gravity is a force that attracts two bodies towards each other. It gives weight to physical objects and is responsible for the movement of planets around the sun.",
] 